# py_exe=/home/lzc/.conda/envs/py38/bin/python3
py_exe=/home/zhicheng.luo/soft/anaconda3/envs/py310/bin/python3

#weights_path=/home/lzc/work/Weights/yolov7-off/v0.1/yolov7.pt
weights_path="/home/zhicheng.luo/Weights/chuwei_best_20240805.pt"

#no dynamic-batch
#$py_exe export.py --weights $weights_path --grid --end2end --simplify --topk-all 100 --iou-thres 0.65 --conf-thres 0.35 --img-size 640 640 --max-wh 640

#for onnx infer
$py_exe export.py --weights $weights_path --grid --dynamic-batch --end2end --simplify --topk-all 100 --iou-thres 0.65 --conf-thres 0.35 --img-size 640 640 --max-wh 640

#for onnx to trt
#$py_exe export.py --weights $weights_path --grid --dynamic-batch --end2end --simplify --topk-all 100 --iou-thres 0.65 --conf-thres 0.35 --img-size 640 640

#--include-nms貌似目前并没啥用
#pip install nvidia-pyindex
#pip install onnx_graphsurgeon --index-url https://pypi.ngc.nvidia.com
#$py_exe export.py --weights $weights_path --grid --dynamic-batch --end2end --simplify --topk-all 100 --iou-thres 0.65 --conf-thres 0.35 --img-size 640 640 --include-nms
